package com.anji.captcha.controller;

import com.anji.captcha.model.common.ResponseModel;
import com.anji.captcha.model.vo.CaptchaVO;
import com.anji.captcha.service.CaptchaService;
import com.anji.captcha.util.StringUtils;
import jakarta.servlet.http.HttpServletRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping({ "/captcha" })
public class CaptchaController {

	@Autowired
	private CaptchaService captchaService;

	public CaptchaController() {
	}

	@PostMapping({ "/get" })
	public ResponseModel get(@RequestBody CaptchaVO data, HttpServletRequest request) {
		assert request.getRemoteHost() != null;

		data.setBrowserInfo(getRemoteId(request));
		return this.captchaService.get(data);
	}

	@PostMapping({ "/check" })
	public ResponseModel check(@RequestBody CaptchaVO data, HttpServletRequest request) {
		data.setBrowserInfo(getRemoteId(request));
		return this.captchaService.check(data);
	}

	public ResponseModel verify(@RequestBody CaptchaVO data, HttpServletRequest request) {
		return this.captchaService.verification(data);
	}

	public static final String getRemoteId(HttpServletRequest request) {
		String xfwd = request.getHeader("X-Forwarded-For");
		String ip = getRemoteIpFromXfwd(xfwd);
		String ua = request.getHeader("user-agent");
		return StringUtils.isNotBlank(ip) ? ip + ua : request.getRemoteAddr() + ua;
	}

	private static String getRemoteIpFromXfwd(String xfwd) {
		if (StringUtils.isNotBlank(xfwd)) {
			String[] ipList = xfwd.split(",");
			return StringUtils.trim(ipList[0]);
		}
		else {
			return null;
		}
	}

}